package com.xu.cloud.codegenerator.convert;

import com.xu.cloud.codegenerator.pojo.domain.TableDO;
import com.xu.cloud.codegenerator.pojo.dto.TableDTO;

import java.util.List;
import java.util.stream.Collectors;

/**
 * TableEntity类转换器
 *
 * @author xuguan
 * @since 2024/1/26
 */
public class TableEntityConverter {

    public static TableDTO convertTableDO(TableDO tableDO) {
        if (tableDO == null) {
            return null;
        }
        TableDTO tableDTO = new TableDTO();
        tableDTO.setTableName(tableDO.getTableName());
        tableDTO.setTableComment(tableDO.getTableComment());
        return tableDTO;
    }

    public static List<TableDTO> convertTableDOList(List<TableDO> tableDOList) {
        if (tableDOList == null) {
            return null;
        }
        return tableDOList.stream()
                .map(TableEntityConverter::convertTableDO)
                .collect(Collectors.toList());
    }
}
